adbench == 0.1.11
torchvision >= 0.14.1
einops >= 0.6.1
torch-geometric == 2.3.1
tensorflow == 2.12.0
